from builder import MODELS


# 使用注册器管理模块
@MODELS.register_module()
class my_ResNet(object):
    def __init__(self, a, b):
        self.a = a
        self.b = b

    def my_print(self):
        print(self.a, self.b)


# 使用注册器管理模块
@MODELS.register_module()
def my_resnet50(a, b):
    return my_ResNet(a, b)
